package org.persistent.impl;

import org.domain.Item;
import org.domain.Product;
import org.persistent.DBUtil;
import org.persistent.ItemDAO;

import java.sql.Connection;
import java.sql.PreparedStatement;
import java.sql.ResultSet;
import java.sql.SQLException;
import java.util.ArrayList;
import java.util.List;
import java.util.Map;

public class ItemDAOImpl implements ItemDAO {
    private static final String UPDATE_INVENTORY_QUANTITY =
            "UPDATE inventory SET qty = qty - ? WHERE itemid = ?";
    private static final String GET_INVENTORY_QUANTITY =
            "SELECT qty FROM inventory WHERE itemid = ?";
    private static final String GET_ITEMLIST_BY_PRODUCT =
            " SELECT I.ITEMID,LISTPRICE,UNITCOST,SUPPLIER AS supplierId,I.PRODUCTID AS \"product.productId\",NAME AS \"product.name\",DESCN AS \"product.description\",CATEGORY AS \"product.categoryId\",STATUS,ATTR1 AS attribute1,ATTR2 AS attribute2,ATTR3 AS attribute3,ATTR4 AS attribute4,ATTR5 AS attribute5 FROM ITEM I, PRODUCT P WHERE P.PRODUCTID = I.PRODUCTID AND I.PRODUCTID = ?";
    private static final String GET_ITEM =
            " SELECT I.ITEMID,LISTPRICE,UNITCOST,SUPPLIER AS supplierId,I.PRODUCTID AS \"product.productId\",NAME AS \"product.name\",DESCN AS \"product.description\",CATEGORY AS \"product.categoryId\",STATUS,ATTR1 AS attribute1,ATTR2 AS attribute2,ATTR3 AS attribute3,ATTR4 AS attribute4,ATTR5 AS attribute5,QTY AS quantity from ITEM I, INVENTORY V, PRODUCT P where P.PRODUCTID = I.PRODUCTID and I.ITEMID = V.ITEMID and I.ITEMID = ?";


    @Override
    public void updateInventoryQuantity(Map<String, Integer> param) {
        try{
            Connection connection = DBUtil.getConnection();
            for (Map.Entry<String, Integer> entry : param.entrySet()) {
                PreparedStatement preparedStatement = connection.prepareStatement(UPDATE_INVENTORY_QUANTITY);
                preparedStatement.setInt(1, Integer.parseInt(entry.getValue().toString()));
                preparedStatement.setString(2, entry.getKey());
                preparedStatement.executeUpdate();
                DBUtil.closePreparedStatement(preparedStatement);
            }
            DBUtil.closeConnection(connection);
        }catch (Exception e)
        {
            e.printStackTrace();
        }
    }

    @Override
    public int getInventoryQuantity(String itemId) {
        int result = 1;
        try{
            Connection connection = DBUtil.getConnection();
            PreparedStatement preparedStatement = connection.prepareStatement(GET_INVENTORY_QUANTITY);
            preparedStatement.setString(1,itemId);
            ResultSet resultSet = preparedStatement.executeQuery();
            if(resultSet.next()){
                result = resultSet.getInt(1);
            }
            DBUtil.closeResultSet(resultSet);
            DBUtil.closePreparedStatement(preparedStatement);
            DBUtil.closeConnection(connection);
        }catch (Exception e)
        {
            e.printStackTrace();
        }

        return result;
    }

    @Override
    public List<Item> getItemListByProduct(String productId) {
        List<Item> itemList = new ArrayList<>();
        try{
            Connection connection = DBUtil.getConnection();
            PreparedStatement preparedStatement = connection.prepareStatement(GET_ITEMLIST_BY_PRODUCT);
            preparedStatement.setString(1,productId);
            ResultSet resultSet = preparedStatement.executeQuery();
            while (resultSet.next())
            {
                Item item = new Item();
                setItem(item,resultSet);
                itemList.add(item);
            }
            DBUtil.closeResultSet(resultSet);
            DBUtil.closePreparedStatement(preparedStatement);
            DBUtil.closeConnection(connection);
        }catch (Exception e)
        {
            e.printStackTrace();
        }
        return itemList;
    }

    @Override
    public Item getItem(String itemId) {
        Item item = null;
        try{
            Connection connection = DBUtil.getConnection();
            PreparedStatement preparedStatement = connection.prepareStatement(GET_ITEM);
            preparedStatement.setString(1,itemId);
            ResultSet resultSet = preparedStatement.executeQuery();
            if(resultSet.next())
            {
                item = new Item();
                setItem(item,resultSet);
            }
            DBUtil.closeResultSet(resultSet);
            DBUtil.closePreparedStatement(preparedStatement);
            DBUtil.closeConnection(connection);
        }catch (Exception e)
        {
            e.printStackTrace();
        }

        return item;
    }

    @Override
    public void setItem(Item item, ResultSet resultSet) throws SQLException {
        item.setItemId(resultSet.getString(1));
        item.setListPrice(resultSet.getBigDecimal(2));
        item.setUnitCost(resultSet.getBigDecimal(3));
        item.setSupplierId(resultSet.getInt(4));
        Product product = new Product();
        product.setProductId(resultSet.getString(5));
        product.setName(resultSet.getString(6));
        product.setDescription(resultSet.getString(7));
        product.setCategoryId(resultSet.getString(8));
        item.setProduct(product);
        item.setStatus(resultSet.getString(9));
        item.setAttribute1(resultSet.getString(10));
        item.setAttribute2(resultSet.getString(11));
        item.setAttribute3(resultSet.getString(12));
        item.setAttribute4(resultSet.getString(13));
        item.setAttribute5(resultSet.getString(14));
    }
}
